#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <fcntl.h>

#define NUM 1024
#define SEP " "
#define SIZE 64

#define NoneRedir 0
#define OutputRedir 1
#define AppendRedir 2
#define InputRedir 3

int redir = NoneRedir;
char *filename = NULL;

char cwd[1024];
int lastcode = 0;
char enval[1024];

// 获取用户名
const char *getUsername()
{
    const char *name = getenv("USER");
    if (name)
        return name;
    return "none";
}
// 获取主机名
const char *getHostname()
{
    const char *hostname = getenv("HOSTNAME");
    if (hostname)
        return hostname;
    return "none";
}
// 获取当前工作目录
const char *getCwd()
{
    const char *cwd = getenv("PWD");
    if (cwd)
        return cwd;
    return "none";
}
// 获取家目录
char *getHomePath()
{
    char *home = getenv("HOME");
    if (home)
        return home;
    else
        return (char *)".";
}

// 打印命令行提示符 + 获取用户指令
int getUserCommand(char *command, int sz)
{
    // 打印命令行提示符
    printf("[%s@%s %s]# ", getUsername(), getHostname(), getCwd());
    // 获取用户输入命令
    char *r = fgets(command, sz, stdin);
    if (r == NULL)
        return 1;
    // 命令规范化
    command[strlen(command) - 1] = '\0'; // 不会存在越界问题
    return strlen(command);
}

// 分割命令字符串
void commandSplit(char *in, char *out[])
{
    int argc = 0;
    out[argc++] = strtok(in, SEP);
    while (out[argc++] = strtok(NULL, SEP))
    {
    };
}

// 执行命令
int excuteCommand(char *argv[])
{
    // 创建子进程执行命令
    pid_t id = fork();
    if (id < 0)
        return -1;
    else if (id == 0) // child
    {
        int fd = 0;
        if (redir == InputRedir) // 输入重定向
        {
            fd = open(filename, O_RDONLY);
            dup2(fd, 0);
        }
        else if (redir == OutputRedir) // 输出重定向
        {
            fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0666);
            dup2(fd, 1);
        }
        else if (redir == AppendRedir) // 追加重定向
        {
            fd = open(filename, O_WRONLY | O_CREAT | O_APPEND, 0666);
            dup2(fd, 1);
        }
        else
        {
            // do nothing
        }
        // 程序替换执行命令
        execvp(argv[0], argv);
        exit(1);
    }
    else // parent
    {
        int status = 0;
        pid_t rid = waitpid(id, &status, 0);
        if (rid > 0)
        {
            lastcode = WEXITSTATUS(status);
        }
    }
    return 0;
}

void cd(char *path)
{
    chdir(path);                 // 改变当前工作目录
    char tmp[1024];              //
    getcwd(tmp, sizeof(tmp));    // 获取当前工作目录
    sprintf(cwd, "PWD=%s", tmp); // 写入到缓冲区中
    putenv(cwd);                 // 导出环境变量
}

// 判断内建命令
int isBuildinCommand(char *argv[])
{
    if (strcmp(argv[0], "cd") == 0) // cd path
    {
        char *path = NULL;
        if (argv[1] == NULL)
            path = getHomePath();
        else
            path = argv[1];
        cd(path);
        return 1;
    }
    else if (strcmp(argv[0], "export") == 0)
    {
        if (argv[1] == NULL)
            return 1;
        strcpy(enval, argv[1]); // 拷贝到缓冲区
        putenv(enval);          // 导出
        return 1;
    }
    else if (strcmp(argv[0], "echo") == 0)
    {
        if (argv[1] == NULL)
        {
            printf("\n");
            return 1;
        }
        else if (*(argv[1]) == '$' && strlen(argv[1]) > 1)
        {
            char *val = argv[1] + 1;
            if (strcmp(val, "?") == 0)
            {
                printf("%d\n", lastcode);
                lastcode = 0;
            }
            else
            {
                const char *env_val = getenv(val);
                if (env_val)
                    printf("%s\n", env_val);
                else
                    printf("\n");
            }
            return 1;
        }
        else
        {
            printf("%s\n", argv[1]);
            return 1;
        }
    }
    // 可以再列举更多内建命令
    return 0;
}

#define SkipSpace(pos)        \
    do                        \
    {                         \
        while (isspace(*pos)) \
            pos++;            \
    } while (0)

// 检查重定向
void checkRedir(char command[], int len)
{
    char *end = command + len - 1; // 从后向前面检查
    char *start = command;
    while (end > start)
    {
        if (*end == '>') // 输出重定向       ls -a -l >> log.txt
        {
            if (*(end - 1) == '>') // 追加重定向
            {
                *(end - 1) = '\0';
                filename = end + 1;
                SkipSpace(filename);
                redir = AppendRedir;
                break;
            }
            else
            {
                *end = '\0';
                filename = end + 1;
                SkipSpace(filename);
                redir = OutputRedir;
                break;
            }
        }
        else if (*end == '<') // 输入重定向
        {
            *end = '\0';
            filename = end + 1;
            SkipSpace(filename);
            redir = InputRedir;
            break;
        }
        else
        {
            end--;
        }
    }
}

int main()
{
    while (1)
    {
        char usercommand[NUM];
        // 打印命令行提示符 + 获取用户指令
        int flag = getUserCommand(usercommand, sizeof(usercommand));
        if (flag <= 0)
            continue;
        // 检测重定向
        checkRedir(usercommand, sizeof(usercommand));
        // 分割命令字符串
        char *argv[SIZE];
        commandSplit(usercommand, argv);
        // 判断内建命令
        int sign = isBuildinCommand(argv);
        if (sign)
            continue;
        // 执行命令
        excuteCommand(argv);
    }
    return 0;
}
